package pdfUtil;
import com.itextpdf.text.Document;
import com.itextpdf.text.DocumentException;
import com.itextpdf.text.pdf.PdfCopy;
import com.itextpdf.text.pdf.PdfImportedPage;
import com.itextpdf.text.pdf.PdfReader;

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.URL;
import java.net.URLConnection;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.ArrayList;
import java.util.List;
import java.util.stream.Collectors;
import java.util.stream.Stream;

/**
 * @see <a href="https://www.cnblogs.com/whatlonelytear/p/9040600.html">用itext合并多个pdf文件【转】【补】</a>
 */
public class PdfDemo {
    public static void main(String[] args) throws IOException {
        // 远程文件
        List<byte[]> fileList = getRemoteFiles();
        // 本地文件
        // List<byte[]> fileList = getLocalFiles("/tmp/pdf");

        // 使用字节os可以让数据写到内存里
        final ByteArrayOutputStream os = new ByteArrayOutputStream();
        // 获取合并的pdf
        mergePdf(os, fileList);
        // 写入硬盘
        Files.write(Paths.get("all.pdf"), os.toByteArray());
    }

    private static List<byte[]> getRemoteFiles() throws IOException {
        // 获取所有url
        String[] urlList = new String[]{
                "http://106.75.225.76:8080/main/test.pdf",
                "http://106.75.225.76:8080/main/test.pdf"
        };
        // 文件列表
        List<byte[]> fileList = new ArrayList<>();
        // 下载所有文件
        for (String path : urlList) {
            // 建立连接
            final URLConnection conn = new URL(path).openConnection();
            // 下载文件
            final InputStream is = conn.getInputStream();
            // 读取进内存,这个时候文件已经完全存在于内存里了
            // 如果不读成 byte 数组的话, 输入流可能会关闭
            // final var bytes = is.readAllBytes(); // jdk 9 提供的方法
            final byte[] bytes = readAllBytes(is);
            fileList.add(bytes);
        }
        return fileList;
    }

    /**
     * 把 inputStream 里的所有数据读取出来
     *
     * @param inputStream 输入流
     * @return 输入流里的所有数据
     * @throws IOException 当读取数据出错时
     */
    private static byte[] readAllBytes(InputStream inputStream) throws IOException {
        final ByteArrayOutputStream os = new ByteArrayOutputStream();
        final byte[] buffer = new byte[1024];
        while (inputStream.read(buffer) != -1) {
            os.write(buffer);
        }
        return os.toByteArray();
    }

    /**
     * 读取指定目录下的所有文件
     *
     * @param location 指定目录, 例如 "/tmp/pdf"
     * @return 指定目录下的所有文件的内容
     * @throws IOException 当读取文件异常时
     */
    private static List<byte[]> getLocalFiles(String location) throws IOException {
        final Path p = Paths.get(location);
        final Stream<Path> list = Files.list(p);
        return list.map(file -> {
            try {
                return Files.readAllBytes(file);
            } catch (IOException e) {
                throw new RuntimeException(e);
            }
        })
                .collect(Collectors.toList());
    }

    /**
     * 合并多个pdf
     *
     * @param os    合并的pdf输出目标流
     * @param files 需要被合并的多个pdf列表
     */
    public static void mergePdf(OutputStream os, List<byte[]> files) {
        Document document = null;
        try {
            // 获取纸张大小并实例化一个新的空文档, 例如 A5 纸
            document = new Document(new PdfReader(files.get(0)).getPageSize(1));
            // 实例化复制工具
            final PdfCopy copy = new PdfCopy(document, os);
            // 打开文档准备写入内容
            document.open();
            // 循环所有pdf文件
            for (byte[] file : files) {
                // 读取pdf
                final PdfReader reader = new PdfReader(file);
                // 获取页数
                final int numberOfPages = reader.getNumberOfPages();
                // pdf的所有页, 从第1页开始遍历, 这里要注意不是0
                for (int i = 1; i <= numberOfPages; i++) {
                    document.newPage();
                    // 把第 i 页读取出来
                    final PdfImportedPage page = copy.getImportedPage(reader, i);
                    // 把读取出来的页追加进输出文件里
                    copy.addPage(page);
                }
            }
        } catch (IOException | DocumentException e) {
            throw new RuntimeException(e);
        } finally {
            if (document != null) {
                document.close();
            }
        }
    }
}

